/*
 * Copyright (c) 2022-2022 Huawei Device Co., Ltd. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this list of
 * conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice, this list
 * of conditions and the following disclaimer in the documentation and/or other materials
 * provided with the distribution.
 *
 * 3. Neither the name of the copyright holder nor the names of its contributors may be used
 * to endorse or promote products derived from this software without specific prior written
 * permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
#include "It_posix_inotify.h"
const int MAX_TEST_NUM = 1000;
const int DECIMAL_CONVERSION = 1000;

static char *inotifyTestFile[] = {"/usr/inotify-test1.txt", "/usr/inotify-test2.txt", "/usr/inotify-test3.txt"};
static int testFileNum = sizeof(inotifyTestFile) / sizeof(inotifyTestFile[0]);

static void TestWrite(int fd)
{
    char writebuf[100] = "test_wirte";
    for (int i = 0; i < MAX_TEST_NUM; i++) {
        size_t size = write(fd, writebuf, (strlen(writebuf) + 1));
        ICUNIT_ASSERT_EQUAL_VOID(size, 11, size); // The string length is 11
    }
}

static void TestRead(int fp)
{
    char readbuf[11];
    for (int i = 0; i < MAX_TEST_NUM; i++) {
        memset(readbuf, 0, sizeof(readbuf));
        size_t size = read(fp, readbuf, sizeof(readbuf));
        ICUNIT_ASSERT_EQUAL_VOID(size, 11, size); // The string length is 11
    }
}

static void TestOpen(const char * pathname)
{
    for (int i = 0; i < MAX_TEST_NUM; i++) {
        int fd = open(pathname, O_WRONLY | O_CREAT);
        ICUNIT_ASSERT_NOT_EQUAL_VOID(fd, -1, fd);
        close(fd);
        int ret = unlink(pathname);
        ICUNIT_ASSERT_EQUAL_VOID(ret, 0, ret);
    }
}

static UINT32 TestWithInotifyWatch()
{
    struct timeval t_start, t_end;
    std::string inotifyTestPath = "/usr/inotify6.txt";
    std::string inotifyDir = "/usr";
    int fd, fp, ret, wd;
    long startTime, endTime;

    fd = open(inotifyTestPath.c_str(), O_WRONLY | O_CREAT);
    ICUNIT_ASSERT_NOT_EQUAL(fd, -1, -1);

    int inotifyFd0 = inotify_init();
    if (inotifyFd0 < 0) {
        ICUNIT_ASSERT_EQUAL(1, 0, inotifyFd0);
    }
    wd = inotify_add_watch(inotifyFd0, inotifyTestPath.c_str(), IN_ACCESS | IN_MODIFY);
    ICUNIT_ASSERT_NOT_EQUAL(wd, -1, wd);
    int inotifyFd1 = inotify_init();
    if (inotifyFd1 < 0) {
        ICUNIT_ASSERT_EQUAL(1, 0, inotifyFd1);
    }
    wd = inotify_add_watch(inotifyFd1, inotifyDir.c_str(), IN_CREATE);
    ICUNIT_ASSERT_NOT_EQUAL(wd, -1, wd);

    printf("Test read/write performance with inotify watch\n");
    gettimeofday(&t_start, NULL);
    startTime = ((long)t_start.tv_sec) * DECIMAL_CONVERSION + (long)t_start.tv_usec / DECIMAL_CONVERSION;
    TestWrite(fd);
    gettimeofday(&t_end, NULL);
    endTime = ((long)t_end.tv_sec) * DECIMAL_CONVERSION + (long)t_end.tv_usec / DECIMAL_CONVERSION;
    printf("Write the file 1000 times takes %ldms\n", (endTime - startTime));
    close(fd);

    fp = open(inotifyTestPath.c_str(), O_RDONLY | O_EXCL);
    gettimeofday(&t_start, NULL);
    startTime = ((long)t_start.tv_sec) * DECIMAL_CONVERSION + (long)t_start.tv_usec / DECIMAL_CONVERSION;
    TestRead(fp);
    gettimeofday(&t_end, NULL);
    endTime = ((long)t_end.tv_sec) * DECIMAL_CONVERSION + (long)t_end.tv_usec / DECIMAL_CONVERSION;
    printf("Read the file 1000 times takes %ldms\n", (endTime - startTime));
    close(fp);

    ret = unlink(inotifyTestPath.c_str());
    ICUNIT_ASSERT_EQUAL(ret, 0, ret);

    gettimeofday(&t_start, NULL);
    startTime = ((long)t_start.tv_sec) * DECIMAL_CONVERSION + (long)t_start.tv_usec / DECIMAL_CONVERSION;
    TestOpen(inotifyTestPath.c_str());
    gettimeofday(&t_end, NULL);
    endTime = ((long)t_end.tv_sec) * DECIMAL_CONVERSION + (long)t_end.tv_usec / DECIMAL_CONVERSION;
    printf("Delete and create a file under the directory 1000 times takes %ldms\n", (endTime - startTime));

    close(inotifyFd0);
    close(inotifyFd1);
    return 0;
}

static UINT32 Testcase(VOID)
{
    struct timeval t_start, t_end;
    std::string inotifyTestPath = "/usr/inotify6.txt";
    std::string inotifyDir = "/usr";
    int fd, fp, ret;
    long startTime, endTime;

    printf("Test read/write performance without inotify watch\n");

    fd = open(inotifyTestPath.c_str(), O_WRONLY | O_CREAT);
    ICUNIT_ASSERT_NOT_EQUAL(fd, -1, -1);
    gettimeofday(&t_start, NULL);
    startTime = ((long)t_start.tv_sec) * DECIMAL_CONVERSION + (long)t_start.tv_usec / DECIMAL_CONVERSION;
    TestWrite(fd);
    gettimeofday(&t_end, NULL);
    endTime = ((long)t_end.tv_sec) * DECIMAL_CONVERSION + (long)t_end.tv_usec / DECIMAL_CONVERSION;
    printf("Write the file 1000 times takes %ldms\n", (endTime - startTime));
    close(fd);

    fp = open(inotifyTestPath.c_str(), O_RDONLY | O_EXCL);
    gettimeofday(&t_start, NULL);
    startTime = ((long)t_start.tv_sec) * DECIMAL_CONVERSION + (long)t_start.tv_usec / DECIMAL_CONVERSION;
    TestRead(fp);
    gettimeofday(&t_end, NULL);
    endTime = ((long)t_end.tv_sec) * DECIMAL_CONVERSION + (long)t_end.tv_usec / DECIMAL_CONVERSION;
    printf("Read the file 1000 times takes %ldms\n", (endTime - startTime));
    close(fp);

    ret = unlink(inotifyTestPath.c_str());
    ICUNIT_ASSERT_EQUAL(ret, 0, ret);

    gettimeofday(&t_start, NULL);
    startTime = ((long)t_start.tv_sec) * DECIMAL_CONVERSION + (long)t_start.tv_usec / DECIMAL_CONVERSION;
    TestOpen(inotifyTestPath.c_str());
    gettimeofday(&t_end, NULL);
    endTime = ((long)t_end.tv_sec) * DECIMAL_CONVERSION + (long)t_end.tv_usec / DECIMAL_CONVERSION;
    printf("Delete and create a file under the directory 1000 times takes %ldms\n", (endTime - startTime));

    return TestWithInotifyWatch();
}

void ItPosixInotify007(void)
{
    TEST_ADD_CASE("ItPosixInotify007", Testcase, TEST_POSIX, TEST_INOTIFY, TEST_LEVEL0, TEST_FUNCTION);
}
